#ifndef RKHumanAttr_h__
#define RKHumanAttr_h__
#include <vector>
#include <string>
#include "vision.h"
struct AttrResult
{

    int Gender = 0;//0: male 1: female
    int Age = 1; //0: 0-18  1: 18-60 2: 60+
    std::vector<int> Appearance; //0-12 ['短袖上衣','长袖上衣',' 短袖外套',' 长袖外套','背心','吊带','短裤','长裤','裙子','短袖连衣裙','长袖连衣裙','马甲连衣裙','吊带连衣裙']
};

class RKHumanAttr
{
public:
    bool Init(std::string modelPath);

    std::shared_ptr<AttrResult> Infer(cv::Mat& image);


private:
    std::shared_ptr<fastdeploy::vision::classification::PaddleClasModel> mAgeGenderModel;
    std::shared_ptr<fastdeploy::vision::detection::PicoDet> mAppearanceModel;
};


#endif //RKHumanAttr_h__